package com.wb.reggie.controller;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.wb.reggie.common.R;
import com.wb.reggie.entity.User;
import com.wb.reggie.service.UserService;
import com.wb.reggie.utils.SMSUtils;
import com.wb.reggie.utils.ValidateCodeUtils;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import javax.servlet.http.HttpSession;
import java.util.Map;

@Slf4j
@RestController
@RequestMapping("/user")
public class UserController {

    @Autowired
    private UserService userService;

    /**
     * 发送手机短信验证码
     * @param user
     * @return
     */
    @PostMapping("/sendMsg")
    public R<String> sendMsg(@RequestBody User user, HttpSession session){
        //获取手机号
        String phone = user.getPhone();
        if(StringUtils.isNotEmpty(phone)){
            //生成随机的6位验证码
            String code = ValidateCodeUtils.generateValidateCode(6).toString();
            /**
             * 现在还没有购买阿里云的服务，不能将验证码通过短信的方式发送到手机上，只能通过控制台将验证码输出，
             * 待会儿登录的时候从控制台拿到一份即可
             *
             *              //调用阿里云提供的短信服务API完成发送短信 (签名，模板code，手机号，动态验证码),
             *             //SMSUtils.sendMessage("","",phone,code);
             */
            log.info("生成的验证码：{}",code);

            //需要将生成的验证码保存到session中，登录的时候需要用到这份验证码
            session.setAttribute("phone",code);

            return R.success("手机验证码短信发送成功！");
        }

        return R.error("短信发送失败");
    }

    /**
     * 移动端用户登录
     * 可以像之前一样用DTO接收，创建一个UserDto继承User，再加入一个code成员变量
     * 也可以用Map接收
     * @return
     */
    @PostMapping("/login")
    public R<User> login(@RequestBody Map map,HttpSession session){
        log.info(map.toString());

        //获取手机号
        String phone = map.get("phone").toString();

        //获取验证码
        String code = map.get("code").toString();

        //从session中保存获取的验证码
        Object codeInSession = session.getAttribute("phone");

        //验证码的比对（在页面提交的验证码 和 保存在session中的验证码进行比对）
        if(codeInSession != null && codeInSession.equals(code)){
            //如果能比对成功，说明登录成功


            //判断当前的手机号对应的用户是否是新用户，如果是新用户则自动完成注册
            //拿该手机号到User表中查询即可
            LambdaQueryWrapper<User> queryWrapper = new LambdaQueryWrapper<>();
            queryWrapper.eq(User::getPhone,phone);
            User user = userService.getOne(queryWrapper);

            String sWord = ValidateCodeUtils.getWord(4);
            String pWord = phone.substring(phone.length() - 4,phone.length());
            if(user == null){//新用户
                user = new User();
                user.setPhone(phone);
                user.setStatus(1);
                user.setName("user_" + sWord + pWord);//默认用户名，user_ + 随机四个字符 + 手机号后四位数
                userService.save(user);//注册
            }

            //登录成功之后，保存一份用户到session
            session.setAttribute("user",user);
            return R.success(user);
        }

        return R.error("短信验证失败");
    }
}
